package com.jyw.system.service.impl;

import cn.hutool.core.lang.UUID;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.jyw.common.constant.CacheConstants;
import com.jyw.common.core.domain.entity.SysDictData;
import com.jyw.common.core.domain.entity.SysDictType;
import com.jyw.common.core.redis.RedisCache;
import com.jyw.common.exception.ServiceException;
import com.jyw.common.utils.DateUtils;
import com.jyw.common.utils.DictUtils;
import com.jyw.common.utils.SecurityUtils;
import com.jyw.system.domain.vo.DictCacheListVo;
import com.jyw.system.mapper.SysDictDataMapper;
import com.jyw.system.service.ISysDictDataService;
import com.jyw.system.service.ISysDictTypeService;
import com.jyw.system.utils.CacheVersionUtils;
import org.springframework.stereotype.Service;
import org.springframework.util.ObjectUtils;

import javax.annotation.Resource;
import java.util.Collection;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 字典 业务层处理
 *
 * @author system
 */
@Service
public class SysDictDataServiceImpl extends ServiceImpl<SysDictDataMapper, SysDictData> implements ISysDictDataService {
    @Resource
    private SysDictDataMapper dictDataMapper;

    @Resource
    private ISysDictTypeService dictTypeService;

    @Resource
    private RedisCache redisCache;

    /**
     * 根据条件分页查询字典数据
     *
     * @param dictData 字典数据信息
     * @return 字典数据集合信息
     */
    @Override
    public List<SysDictData> selectDictDataList(SysDictData dictData) {
        return dictDataMapper.selectDictDataList(dictData);
    }

    /**
     * 根据字典类型和字典键值查询字典数据信息
     *
     * @param dictType  字典类型
     * @param dictValue 字典键值
     * @return 字典标签
     */
    @Override
    public String selectDictLabel(String dictType, String dictValue) {
        return dictDataMapper.selectDictLabel(dictType, dictValue);
    }

    /**
     * 根据字典数据ID查询信息
     *
     * @param dictCode 字典数据ID
     * @return 字典数据
     */
    @Override
    public SysDictData selectDictDataById(Long dictCode) {
        return dictDataMapper.selectDictDataById(dictCode);
    }

    /**
     * 批量删除字典数据信息
     *
     * @param dictCodes 需要删除的字典数据ID
     */
    @Override
    public void deleteDictDataByIds(Long[] dictCodes) {
        for (Long dictCode : dictCodes) {
            SysDictData data = selectDictDataById(dictCode);
            dictDataMapper.deleteDictDataById(dictCode);
            List<SysDictData> dictDatas = dictDataMapper.selectDictDataByType(data.getDictType());
            DictUtils.setDictCache(data.getDictType(), dictDatas);
            CacheVersionUtils.dictVersionUp();
        }
    }

    /**
     * 新增保存字典数据信息
     *
     * @param data 字典数据信息
     * @return 结果
     */
    @Override
    public int insertDictData(SysDictData data) {
        int row = dictDataMapper.insertDictData(data);
        if (row > 0) {
            List<SysDictData> dictDatas = dictDataMapper.selectDictDataByType(data.getDictType());
            DictUtils.setDictCache(data.getDictType(), dictDatas);
            CacheVersionUtils.dictVersionUp();
        }
        return row;
    }

    /**
     * 修改保存字典数据信息
     *
     * @param data 字典数据信息
     * @return 结果
     */
    @Override
    public int updateDictData(SysDictData data) {
        if(data.getParentId()!=null&&data.getParentId().equals(data.getDictCode())){
            throw new ServiceException("上级不能选择当前字典");
        }
        int row = dictDataMapper.updateDictData(data);
        if (row > 0) {
            List<SysDictData> dictDatas = dictDataMapper.selectDictDataByType(data.getDictType());
            DictUtils.setDictCache(data.getDictType(), dictDatas);
            CacheVersionUtils.dictVersionUp();
        }
        return row;
    }

    @Override
    public SysDictData getTopLevelFaultType(String dictValue) {
        SysDictData data = dictDataMapper.getTopLevelFaultType(dictValue);
        return data;
    }

    @Override
    public List<DictCacheListVo> getCache() {
        Collection<String> keys = redisCache.keys(DictUtils.getCacheKey("*"));
        return keys.stream().map(data -> new DictCacheListVo(getKey(data), DictUtils.getDictCache(getKey(data)))).collect(Collectors.toList());
    }

    @Override
    public String getDictValueSave(String key, String label) {
        if (ObjectUtils.isEmpty(label)) {
            return null;
        }
        SysDictType sysDictType = dictTypeService.selectDictTypeByType(key);
        if (sysDictType.getDictLevel().equals(1)) {
            //多层级
            String[] labelList = label.split("/");
            //声明父级id
            Long parentId = null;
            String value = null;
            for (String labelStr : labelList) {
                SysDictData data = findValueByLabel(key, parentId, labelStr);
                //赋值
                parentId = data.getDictCode();
                value = data.getDictValue();
            }
            return value;
        } else {
            return findValueByLabel(key, null, label).getDictValue();
        }
    }

    @Override
    public String getDictValue(String key, String label) {
        if (ObjectUtils.isEmpty(label)) {
            return null;
        }
        SysDictType sysDictType = dictTypeService.selectDictTypeByType(key);
        if (sysDictType.getDictLevel().equals(1)) {
            //多层级
            String[] labelList = label.split("/");
            //声明父级id
            Long parentId = null;
            String value = null;
            for (String labelStr : labelList) {
                SysDictData data = findValueByLabelNotSave(key, parentId, labelStr);
                if (ObjectUtils.isEmpty(data)) {
                    return null;
                }
                //赋值
                parentId = data.getDictCode();
                value = data.getDictValue();
            }
            return value;
        } else {
            SysDictData data = findValueByLabelNotSave(key, null, label);
            return ObjectUtils.isEmpty(data) ? null : data.getDictValue();
        }
    }

    @Override
    public List<SysDictData> listByDictType(String dictType) {
        QueryWrapper<SysDictData> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().eq(SysDictData::getDictType, dictType);
        List<SysDictData> dictList = this.list(queryWrapper);
        return dictList;
    }

    /**
     * 查询key下面是否存在该label 如果不存在那么需要新增
     *
     * @param key
     * @param parentId
     * @param label
     * @return
     */
    private SysDictData findValueByLabelNotSave(String key, Long parentId, String label) {
        //查询key parentId下是否存在该label
        LambdaQueryChainWrapper<SysDictData> queryChainWrapper = lambdaQuery()
                .eq(SysDictData::getDictType, key)
                .eq(SysDictData::getDictLabel, label);
        if (ObjectUtils.isEmpty(parentId)) {
            queryChainWrapper.isNull(SysDictData::getParentId);
        } else {
            queryChainWrapper.eq(SysDictData::getParentId, parentId);
        }
        SysDictData data = queryChainWrapper.one();
        return data;
    }

    /**
     * 查询key下面是否存在该label 如果不存在那么需要新增
     *
     * @param key
     * @param parentId
     * @param label
     * @return
     */
    private SysDictData findValueByLabel(String key, Long parentId, String label) {
        //查询key parentId下是否存在该label
        LambdaQueryChainWrapper<SysDictData> queryChainWrapper = lambdaQuery()
                .eq(SysDictData::getDictType, key)
                .eq(SysDictData::getDictLabel, label);
        if (ObjectUtils.isEmpty(parentId)) {
            queryChainWrapper.isNull(SysDictData::getParentId);
        } else {
            queryChainWrapper.eq(SysDictData::getParentId, parentId);
        }
        SysDictData data = queryChainWrapper.one();
        if (ObjectUtils.isEmpty(data)) {
            //不存在则新增
            data = new SysDictData();
            data.setDictLabel(label);
            data.setDictValue(UUID.fastUUID().toString(true));
            data.setDictType(key);
            data.setParentId(parentId);
            data.setStatus("0");
            data.setCreateBy(SecurityUtils.getUsername());
            data.setCreateTime(DateUtils.getNowDate());
            data.setRemark("系统自动创建");
            save(data);
//            缓存升级
            List<SysDictData> dictDatas = dictDataMapper.selectDictDataByType(data.getDictType());
            DictUtils.setDictCache(data.getDictType(), dictDatas);
            CacheVersionUtils.dictVersionUp();
        }
        return data;
    }


    private String getKey(String redisKey) {
        return redisKey.substring(CacheConstants.SYS_DICT_KEY.length());
    }
}
